#! /usr/bin/python
# -*- coding: utf-8 -*-
import os
import urllib

def save_img(img_url,file_name,file_path='book/img'):
    #保存图片到磁盘文件夹 file_path中，默认为当前脚本运行目录下的 book\img文件夹
    try:
        if not os.path.exists(file_path):
            print '文件夹',file_path,'不存在，重新建立'
            #os.mkdir(file_path)
            os.makedirs(file_path)
    except IOError as e:
        print '文件操作失败',e
    #获得图片后缀
    file_suffix = os.path.splitext(img_url)[1]
    #拼接图片名（包含路径）
    filename = '{}{}{}{}'.format(file_path,os.sep,file_name,file_suffix)
   #下载图片，并保存到文件夹中
    urllib.urlretrieve(img_url,filename=filename)


if __name__ == '__main__':
	image_url = 'https://img.gushiwen.org/authorImg/libai.jpg'
	save_img(image_url, 'libai')